#!/usr/bin/python
# -*- coding:utf-8 -*-
# @FileName : DL5_test4_1.py
# Author    : myh

import torch
import torch.nn.functional as F
from torch import nn


# class MyLinear(nn.Module):
#     def __init__(self, A , B ):
#         super().__init__()
#         self.weight = nn.Parameter(torch.randn(A, B))
#     def forward(self, X):
#         y = 0
#         c, = X.shape
#         for i in range(c):
#             x1 = X[:][i]
#             for j in range(c):
#                 x2 = X[:][j]
#                 y_1 =  torch.matmul(x1,x2)
#                 print(self.weight.data[x1,x2])
#                 y = y + torch.matmul(y_1, self.weight.data[])
#                 print(y)
#         return y

X = [[1,2,3,4,5], [6,7,8,9,10]]
print(X)
print(torch.tensor(X)[:,3])
